package com.system.neural.config;
import lombok.Data;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Configuration;

/**
 * ClassName: ClassModelConfig
 * FileName: ClassModelConfig.java
 * Description: 分类模型配置类
 * History:
 * 版本号 			作者 			日期       				简介
 * 1.0				znlccy		    2021/10/4 12:19		    create
 */
@Configuration
@Data
public class ClassModelConfig {

    /**
     * 类别数值
     * 例如: 初始化为{'军事'，‘经济’，‘政治’}
     */
    private String[] categories;

    /**
     * 训练样本的主目录，可以是相对路径，默认是data/training
     * 如果有上面的类别，那么data/training目录中必须包含
     * 军事/ 经济/ 政治/ 这三个目录，分别存放对应类别的数据
     */
    @Value("train.root.dir")
    private String trainRootDir = "data/training";

    /**
     * 训练好模型之后就需要保持模型，这就是模型保持的目录及文件
     * 训练模型的目录和文件data/model/model.class
     */
    @Value("train.model.file")
    private String trainModelFile = "data/model/model.class";
}
